04fbbd1f4e2bfc6e930a32be768195e53c4366db,kafka-scheduler/src/test/java/com/mesosphere/dcos/kafka/scheduler/RecoverySchedulerTest.java,RecoverySchedulerTest,testReplaceNonLastBroker,#,80

Before Change


        when(stateStore.fetchTasksNeedingRecovery()).thenReturn(Arrays.asList(replaceTaskInfo));

        DefaultRecoveryScheduler recoveryScheduler = getTestKafkaRecoveryScheduler();
        List<Protos.OfferID> acceptedOfferIds = recoveryScheduler.resourceOffers(
                driver,
                Arrays.asList(getTestOfferSufficientForNewBroker()),
                Optional.empty());
        Assert.assertEquals(1, acceptedOfferIds.size());
        Assert.assertEquals(KafkaTestUtils.testOfferId, acceptedOfferIds.get(0).getValue());
        verify(driver, times(1)).acceptOffers(

After Change


        when(stateStore.fetchStatus(any())).thenReturn(Optional.of(statuses.get(0)));

        DefaultRecoveryPlanManager recoveryPlanManager = getTestKafkaRecoveryPlanManager();
        DefaultPlanScheduler planScheduler = getTestKafkaScheduler();

        List<Protos.OfferID> acceptedOfferIds = new ArrayList<>(
                planScheduler.resourceOffers(
                      driver,
                      Arrays.asList(getTestOfferSufficientForNewBroker()),
                      recoveryPlanManager.getCandidates(Collections.emptyList()))
        );

        Assert.assertEquals(1, acceptedOfferIds.size());
        Assert.assertEquals(KafkaTestUtils.testOfferId, acceptedOfferIds.get(0).getValue());